"""
美国、日本、印度三个国家确诊人数对比
JSON 在线格式地址：ab173.com
"""
import json

from pyecharts.charts import Line
from pyecharts.options import TitleOpts, LabelOpts
import os

# 打开文件，读取数据
f_us = open('D:\Document\python学习\资料\可视化案例数据\折线图数据\美国.txt', 'r', encoding='UTF-8')
f_jp = open('D:\Document\python学习\资料\可视化案例数据\折线图数据\日本.txt', 'r', encoding='UTF-8')
f_in = open('D:\Document\python学习\资料\可视化案例数据\折线图数据\印度.txt', 'r', encoding='UTF-8')

us_data = f_us.read()
jp_data = f_jp.read()
in_data = f_in.read()

# 处理不规范数据
us_data = us_data.replace('jsonp_1629344292311_69436(', '')[:-2]
jp_data = jp_data.replace('jsonp_1629350871167_29498(', '')[:-2]
in_data = in_data.replace('jsonp_1629350745930_63180(', '')[:-2]

# JSON 格式转为 dict
us_dict_data = json.loads(us_data)
jp_dict_data = json.loads(jp_data)
in_dict_data = json.loads(in_data)

# 获取 trend key
us_trend_data = us_dict_data['data'][0]['trend']
jp_trend_data = jp_dict_data['data'][0]['trend']
in_trend_data = in_dict_data['data'][0]['trend']

# 获取日期数据，用于 x 轴
us_x_data = us_trend_data['updateDate'][:314]
jp_x_data = jp_trend_data['updateDate'][:314]
in_x_data = in_trend_data['updateDate'][:314]

# 获取确诊数据，用于 y 轴
us_y_data = us_trend_data['list'][0]['data'][:314]
jp_y_data = jp_trend_data['list'][0]['data'][:314]
in_y_data = in_trend_data['list'][0]['data'][:314]

# 生成折线图对象
line = Line()

# 添加 x 轴数据
line.add_xaxis(us_x_data)

# 添加 y 轴数据
line.add_yaxis('美国确诊人数', us_y_data, label_opts=LabelOpts(is_show=False))
line.add_yaxis('日本确诊人数', jp_y_data, label_opts=LabelOpts(is_show=False))
line.add_yaxis('印度确诊人数', in_y_data, label_opts=LabelOpts(is_show=False))

# 设置全局选项
line.set_global_opts(
    # 标题
    title_opts=TitleOpts(title='美日印三国确诊人数对比', pos_left='center', pos_bottom='1%')
)

# 生成折线图 render.html
line.render()

# 关闭文件
f_us.close()
f_jp.close()
f_in.close()

# 打开折线图
os.system('render.html')
